package week_02;

import java.util.HashSet;
import java.util.Map;
import java.util.Set;

/**
 * 874. 模拟行走机器人
 * 机器人在一个无限大小的 XY 网格平面上行走，从点(0, 0) 处开始出发，面向北方。该机器人可以接收以下三种类型的命令 commands ：
 * <p>
 * -2 ：向左转90 度
 * -1 ：向右转 90 度
 * 1 <= x <= 9 ：向前移动x个单位长度
 * 在网格上有一些格子被视为障碍物obstacles 。第 i个障碍物位于网格点 obstacles[i] = (xi, yi) 。
 * <p>
 * 机器人无法走到障碍物上，它将会停留在障碍物的前一个网格方块上，但仍然可以继续尝试进行该路线的其余部分。
 * <p>
 * 返回从原点到机器人所有经过的路径点（坐标为整数）的最大欧式距离的平方。（即，如果距离为 5 ，则返回 25 ）
 * <p>
 * <p>
 * 注意：
 * <p>
 * 北表示 +Y 方向。
 * 东表示 +X 方向。
 * 南表示 -Y 方向。
 * 西表示 -X 方向。
 *
 * <p>
 * 示例 1：
 * <p>
 * 输入：commands = [4,-1,3], obstacles = []
 * 输出：25
 * 解释：
 * 机器人开始位于 (0, 0)：
 * 1. 向北移动 4 个单位，到达 (0, 4)
 * 2. 右转
 * 3. 向东移动 3 个单位，到达 (3, 4)
 * 距离原点最远的是 (3, 4) ，距离为 32 + 42 = 25
 * 示例2：
 * <p>
 * 输入：commands = [4,-1,4,-2,4], obstacles = [[2,4]]
 * 输出：65
 * 解释：机器人开始位于 (0, 0)：
 * 1. 向北移动 4 个单位，到达 (0, 4)
 * 2. 右转
 * 3. 向东移动 1 个单位，然后被位于 (2, 4) 的障碍物阻挡，机器人停在 (1, 4)
 * 4. 左转
 * 5. 向北走 4 个单位，到达 (1, 8)
 * 距离原点最远的是 (1, 8) ，距离为 12 + 82 = 65
 * <p>
 */
public class RobotSim_874 {
    // N E S W
    private int[] dx = new int[]{0, 1, 0, -1};
    private int[] dy = new int[]{1, 0, -1, 0};

    public int robotSim(int[] commands, int[][] obstacles) {
        Set<String> obstacleSet = new HashSet<>();
        for (int[] item : obstacles) {
            obstacleSet.add(hashMac(item[0], item[1]));
        }
        int ans = 0;
        int dir = 0;
        int x = 0;
        int y = 0;
        for (int cmd : commands) {
            if (cmd > 0) {
                for (int i = 1; i <= cmd; i++) {
                    int nx = x + dx[dir];
                    int ny = y + dy[dir];
                    if (obstacleSet.contains(hashMac(nx, ny))) break;
                    x = nx;
                    y = ny;
                    ans = Math.max(ans, nx * nx + ny * ny);
                }
            } else if (cmd == -1) {
                // 0->1->2->3->0
                // N->E->S->W->N
                // 右转（顺时针）
                dir = (dir + 1) % 4;
            } else {
                // 左转，避免负数，加一个mod数
                dir = (dir - 1 + 4) % 4;
            }
        }
        return ans;
    }

    public String hashMac(int x, int y) {
        return x + "," + y;
    }
}
